package com.eduagent.xwqeduagent.service;

import com.eduagent.xwqeduagent.utils.CosUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.util.UUID;

/**
 * 医疗文件服务类
 */
@Service
@Slf4j
public class MedicalFileService {

    private static final String MEDICAL_RECORD_DIR = "medical-records";
    private static final String TEMP_DIR = System.getProperty("java.io.tmpdir");

    @Autowired
    private CosUtils cosUtils;

    /**
     * 上传医疗记录HTML文件到COS
     *
     * @param file HTML文件
     * @return 文件访问URL
     */
    public String uploadMedicalRecordFile(MultipartFile file) {
        validateHtmlFile(file);
        return cosUtils.uploadFile(file, MEDICAL_RECORD_DIR);
    }

    /**
     * 将HTML字符串保存为文件并上传到COS
     *
     * @param htmlContent HTML内容
     * @return 文件访问URL
     */
    public String uploadMedicalRecordContent(String htmlContent) {
        try {
            // 创建临时文件
            String fileName = UUID.randomUUID() + ".html";
            File tempFile = new File(TEMP_DIR, fileName);
            
            // 写入HTML内容
            try (FileOutputStream fos = new FileOutputStream(tempFile)) {
                fos.write(htmlContent.getBytes(StandardCharsets.UTF_8));
            }
            
            // 上传到COS
            String fileUrl = cosUtils.uploadFile(tempFile, MEDICAL_RECORD_DIR);
            
            // 删除临时文件
            if (!tempFile.delete()) {
                log.warn("Failed to delete temporary file: {}", tempFile.getAbsolutePath());
            }
            
            return fileUrl;
        } catch (IOException e) {
            log.error("Error creating temporary file: {}", e.getMessage(), e);
            throw new RuntimeException("创建临时文件失败: " + e.getMessage(), e);
        }
    }

    /**
     * 从COS删除医疗记录文件
     *
     * @param fileUrl 文件URL
     * @return 是否删除成功
     */
    public boolean deleteMedicalRecordFile(String fileUrl) {
        return cosUtils.deleteFile(fileUrl);
    }

    /**
     * 从COS读取HTML文件内容
     *
     * @param fileUrl 文件URL
     * @return HTML内容
     */
    public String readHtmlFromCos(String fileUrl) {
        try {
            // 从URL中提取key
            String key = cosUtils.getFileKeyFromUrl(fileUrl);
            
            // 检查文件是否存在
            if (!cosUtils.doesObjectExist(key)) {
                throw new RuntimeException("文件不存在: " + fileUrl);
            }
            
            // 直接获取文件内容
            return cosUtils.getFileContentAsString(key);
        } catch (Exception e) {
            log.error("Error reading file: {}", e.getMessage(), e);
            throw new RuntimeException("读取文件失败: " + e.getMessage(), e);
        }
    }

    /**
     * 验证文件是否为HTML文件
     *
     * @param file 文件
     */
    private void validateHtmlFile(MultipartFile file) {
        String originalFilename = file.getOriginalFilename();
        if (originalFilename == null || !originalFilename.toLowerCase().endsWith(".html")) {
            throw new IllegalArgumentException("只支持HTML文件");
        }
        
        String contentType = file.getContentType();
        if (contentType == null || !contentType.contains("html")) {
            throw new IllegalArgumentException("文件类型不正确，需要HTML文件");
        }
    }

    /**
     * 从COS下载HTML文件到本地
     *
     * @param fileUrl 文件URL
     * @param localFile 本地文件
     * @return 是否下载成功
     */
    public boolean downloadMedicalRecordFile(String fileUrl, File localFile) {
        try {
            // 从URL中提取key
            String key = cosUtils.getFileKeyFromUrl(fileUrl);
            
            // 检查文件是否存在
            if (!cosUtils.doesObjectExist(key)) {
                throw new RuntimeException("文件不存在: " + fileUrl);
            }
            
            // 下载文件
            return cosUtils.downloadFile(key, localFile);
        } catch (Exception e) {
            log.error("Error downloading file: {}", e.getMessage(), e);
            return false;
        }
    }
    
    /**
     * 从COS下载HTML文件到临时目录
     *
     * @param fileUrl 文件URL
     * @return 临时文件
     */
    public File downloadMedicalRecordToTemp(String fileUrl) {
        try {
            // 从URL中提取key
            String key = cosUtils.getFileKeyFromUrl(fileUrl);
            
            // 检查文件是否存在
            if (!cosUtils.doesObjectExist(key)) {
                throw new RuntimeException("文件不存在: " + fileUrl);
            }
            
            // 创建临时文件
            String fileName = UUID.randomUUID() + ".html";
            File tempFile = new File(TEMP_DIR, fileName);
            
            // 下载文件
            if (cosUtils.downloadFile(key, tempFile)) {
                return tempFile;
            } else {
                throw new RuntimeException("下载文件失败: " + fileUrl);
            }
        } catch (Exception e) {
            log.error("Error downloading file: {}", e.getMessage(), e);
            throw new RuntimeException("下载文件失败: " + e.getMessage(), e);
        }
    }
} 